979B - Treasure Hunt - CodeForces Solution

#include "bits/stdc++.h"
#include <iostream>
using namespace std;

int main() {
    int t,mx=0;
    char mxC;
    cin >>t;
    string s;
    int arr[3];
    map<char,int>mp;
    for(int i=0;i<3;i++){
        mp.clear();
        mx=0;
        cin >>s;
        for(auto x:s){
            mp[x]++;
        }
        for(auto &x:mp){
            mx= max(x.second,mx);
        }
        if(mx+t>s.length()){
            arr[i]=s.length();
            if (t==1)
                arr[i]--;
        }
        else
            arr[i]=mx+t;
    }
    if(arr[0]>arr[1] && arr[0]>arr[2])
        cout<<"Kuro";
    else if(arr[1]>arr[0] && arr[1]>arr[2])
        cout<<"Shiro";
    else if(arr[2]>arr[1] && arr[2]>arr[0])
        cout<<"Katie";
    else if(arr[0]==arr[1] || arr[0]==arr[2] || arr[1]==arr[2])
        cout<<"Draw";
}